from selenium.webdriver.common.by import By

from weekend.wework.base.wework_page import WeWorkPage
from weekend.wework.page.catogory import Category


class MainPage(WeWorkPage):
    def goto_category(self):
        # 前往通讯录界面
        self.find_and_click(By.XPATH, "//*[@text='通讯录']")
        return Category(self.driver)
